package com.atguigu.mapReduce.partitioner;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Partitioner;

public class ProvincePartitioner extends Partitioner<Text, FlowBean> {
    @Override
    public int getPartition(Text text, FlowBean flowBean, int numPartitions) {
        String phone = text.toString();
        String prePhone = phone.substring(0, 3);
        int partitionNum ;
        if("136".equals(prePhone)){
            partitionNum = 0;
        }
        else if("137".equals(prePhone)){
            partitionNum = 1;
        }
        else if("138".equals(prePhone)){
            partitionNum=2;
        }else if("139".equals(prePhone)){
            partitionNum=3;
        }
        else {
            partitionNum=4;
        }
        return partitionNum;
    }
}
